//
// Created by Tunm-Air13 on 2024/3/14.
//
#include "semantic_classify/semantic_cls.h"
#include "log.h"

using namespace solex;


std::vector<std::string> semantics = {
        "给我推荐一周的菜单，每天3菜1汤，不要重复",
        "今天想吃牛毽子肉，给我推荐几个做法",
        "周一和周三的菜单里要一些海鲜，做法清蒸",
        "周二和周五健身，帮我搭配成减脂餐",
        "周六有上海朋友来家里做客，帮我生成一份6人的菜单，口味偏上海口味一些，菜单里要有帝王蟹和牛肉",
        "红膏炝蟹怎么做？",
        "八宝鸭怎么做？",
        "开始做饭",
        "帮我设定30分钟的闹钟",
        "打开垃圾处理器",
        "开启下水管疏通",
        "今天天气怎么样？",
        "开启下水管除菌",
        "卤牛肉用牛的什么部位更合适？",
        "请打开100毫升的水",
        "水温调节45度",
};

int main() {

    std::string folderPath = "../pack";

    SemanticCls cls;
    cls.Initialization(folderPath);

    std::string text = "帮我放500毫升水";

    SemanticCls::SemanticPredictResult result;
    cls.Predict(text, result);

    LOGD("%d -> %s : %f", result.idx, result.labelCh.c_str(), result.probability);
    LOGD("cut: %s", result.cutResult.c_str());

    return 0;
}